# -*- coding: utf-8 -*-
# @Time    : 2021/10/26 16:17
# @Author  : huangwei
# @File    : config_args.py
# @Software: PyCharm
from easydict import EasyDict as edict

__C = edict()

args = __C

# params for prediction engine
__C.use_gpu = False

# params for text detector
__C.det_algorithm = 'DB'
__C.det_model_dir = './inference/det_db'

# DB parmas
__C.det_db_thresh = 0.3
__C.use_dilation = False
__C.det_db_box_thresh = 0.3
__C.det_db_unclip_ratio = 1.6

# params for text recognizer
__C.rec_batch_num = 6
__C.rec_char_type = 'ch'
__C.enable_mkldnn = False
__C.rec_algorithm = 'CRNN'
__C.use_space_char = True
__C.rec_image_shape = "3, 32, 320"
__C.rec_model_dir = './inference/rec_crnn'
__C.rec_char_dict_path = "./ppocr/utils/ppocr_keys_v1.txt"

__C.ir_optim = True
__C.use_tensorrt = False
__C.use_fp16 = False
__C.gpu_mem = 500
__C.det_limit_side_len = 960
__C.det_limit_type = 'max'
__C.max_batch_size = 10
__C.max_text_length = 25
__C.vis_font_path = "./doc/fonts/simfang.ttf"
__C.drop_score = 0.5
__C.use_pdserving = False
